package com.crrcdt.mail.config;


import com.crrcdt.mail.constant.MailConstants;
import com.crrcdt.mail.entity.MailSendLog;
import com.crrcdt.mail.service.MailSendLogService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.amqp.core.Binding;
import org.springframework.amqp.core.BindingBuilder;
import org.springframework.amqp.core.DirectExchange;
import org.springframework.amqp.core.Queue;
import org.springframework.amqp.rabbit.connection.CachingConnectionFactory;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class RabbitConfig {
    public final static Logger logger = LoggerFactory.getLogger(RabbitConfig.class);
    //连接工厂
    @Autowired
    CachingConnectionFactory cachingConnectionFactory;
    @Autowired
    MailSendLogService mailSendLogService;

    //对rabbitTemplate做一个定制操作
    @Bean
    RabbitTemplate rabbitTemplate() {
        RabbitTemplate rabbitTemplate = new RabbitTemplate(cachingConnectionFactory);
//        消息只要被 rabbitmq broker 接收到就会触发 confirmCallback 回调 。
// correlationData：对象内部只有一个 id 属性，用来表示当前消息的唯一性。
//ack：消息投递到 broker 的状态， true表示成功。
//cause：表示投递失败的原因。
//但消息被 broker 接收到只能表示已经到达 MQ服务器，并不能保证消息一定会被投递到目标 queue 里。所以接下来需要用到 returnCallback 。
        //TODO 1
        // 1、确认模式 p(public)->b(broker)
        rabbitTemplate.setConfirmCallback((data, ack, cause) -> {
            String msgId = data.getId();
            //TODO 3、ACK机制 q(queue)->c(consumer)（这里只是演示一下，这个是再消费端消费了就会有ack机制，比事务要高效250倍数）
            if (ack) {
                logger.info(msgId + ":消息发送成功");
                final MailSendLog mailSendLog = mailSendLogService.getById(msgId);
                mailSendLog.setStatus(1);
                mailSendLogService.updateById(mailSendLog);//修改数据库中的记录，消息投递成功
            } else {
                logger.info(msgId + ":消息发送失败");
            }
        });
        //TODO 2、未投递到queue 退回模式 e(exchange)->q
//        如果消息未能投递到目标 queue 里将触发回调 returnCallback ，一旦向 queue 投递消息未成功，这里一般会记录下当前消息的详细投递数据，方便后续做重发或者补偿等操作。
        rabbitTemplate.setReturnCallback((msg, repCode, repText, exchange, routingkey) -> {
            logger.info("消息发送失败");
        });
        return rabbitTemplate;
    }

    @Bean
    Queue mailQueue() {
        return new Queue(MailConstants.MAIL_QUEUE_NAME, true);
    }

    @Bean
    DirectExchange mailExchange() {
        return new DirectExchange(MailConstants.MAIL_EXCHANGE_NAME, true, false);
    }

    @Bean
    Binding mailBinding() {
        return BindingBuilder.bind(mailQueue()).to(mailExchange()).with(MailConstants.MAIL_ROUTING_KEY_NAME);
    }

}
